package de.hdu.pms.dao;

import java.util.Calendar;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.List;

import org.apache.log4j.Logger;
import org.junit.Test;

import de.hdu.pms.model.Arbeitstag;
import de.hdu.pms.model.Basis;
import de.hdu.pms.model.Mitarbeiter;
import de.hdu.pms.model.Projekte;
import de.hdu.pms.model.Projektzeit;

public class ProjektzeitDaoTest extends AbstractDataAccessTest {

	public static Logger Log = Logger.getLogger(PausenDaoTest.class);

	private MitarbeiterDao mitarbeiterDao = null;
	private ArbeitstagDao arbeitstagDao = null;
	private BasisDao basisDao = null;
	private ProjektzeitDao projektzeitDao = null;
	private ProjekteDao projekteDao = null;

	public void setMitarbeiterDao(MitarbeiterDao mitarbeiterDao) {
		this.mitarbeiterDao = mitarbeiterDao;
	}

	public void setArbeitstagDao(ArbeitstagDao arbeitstagDao) {
		this.arbeitstagDao = arbeitstagDao;
	}

	public void setBasisDao(BasisDao basisDao) {
		this.basisDao = basisDao;
	}

	public void setProjektzeitDao(ProjektzeitDao projektzeitDao) {
		this.projektzeitDao = projektzeitDao;
	}

	public void setProjekteDao(ProjekteDao projekteDao) {
		this.projekteDao = projekteDao;
	}

	public static void setLog(Logger log) {
		Log = log;
	}

	@Test
	public void testFindById() {
		Log.info("UnitTest - Projektzeit - FindById");
		this.makeZeit(false);
		List<Projektzeit> alle = projektzeitDao.findAll();
		for(Projektzeit p : alle){
			Projektzeit found = projektzeitDao.findById(p.getId());
			assertEquals(p.getId(), found.getId());
			assertEquals(p.getProjekt_von(), found.getProjekt_von());
			assertEquals(p.getProjekt_bis(), found.getProjekt_bis());
		}
		this.destroyZeit();
	}

	@SuppressWarnings("deprecation")
	@Test
	public void testSave() {
		Log.info("UnitTest - Projektzeit - Save");
		this.makeZeit(false);
		int rows = countRowsInTable("projektzeit");
		assertEquals("Es darf nur ein Datensatz vorhanden sein", 1, rows);
		this.destroyZeit();
	}

	@Test
	public void testFindAll() {
		Log.info("UnitTest - Projektzeit - FindAll");
		this.testFindById();
	}

	@Test
	public void testDelete() {
		Log.info("UnitTest - Projektzeit - Delete");
		Projektzeit zeit = this.makeZeit(false);
		try{
			projektzeitDao.delete(zeit);
		}catch (Exception e) {
			Log.info("Exception: " + e.getCause());
		}
		List<Projektzeit> alle = projektzeitDao.findAll();
		assertEquals(0, alle.size());
		this.destroyZeit();
	}

	@Test
	public void testFindCheckedIn() {
		Log.info("UnitTest - Projektzeit - FindCheckedIn");
		Projektzeit zeit = this.makeZeit(true);
		Projektzeit checkedIn = projektzeitDao.findCheckedIn(zeit.getArbeitstag().getId());
		assertEquals(zeit.getId(), checkedIn.getId());
		this.destroyZeit();
	}

	@Test
	public void testFindCron() {
		Log.info("UnitTest - Projektzeit - FindCron");
		this.makeZeit(true);
		Calendar calendar = new GregorianCalendar();
		calendar.set(Calendar.DATE, 9);
		calendar.set(Calendar.MONTH, 9);
		calendar.set(Calendar.YEAR, 2012);
		calendar.set(Calendar.HOUR_OF_DAY, 18);
		calendar.set(Calendar.MINUTE, 0);
		calendar.set(Calendar.SECOND, 0);
		calendar.set(Calendar.MILLISECOND, 0);
		Date datumCron = calendar.getTime();
		calendar.add(Calendar.DATE, 1);
		List<Projektzeit> cron = projektzeitDao.findCron(calendar.getTime());
		for (Projektzeit p : cron) {
			assertEquals(datumCron, p.getArbeitstag().getAtag_datum());
			assertEquals(null, p.getProjekt_bis());
		}
		this.destroyZeit();
		
	}
	
	@SuppressWarnings("deprecation")
	private Projektzeit makeZeit(boolean flag){
		String[] tables = { "projektzeit" };
		String[] tatag = { "arbeitstag" };
		String[] tmitarbeiter = { "mitarbeiter" };
		String[] tbasis = { "basis" };
		String[] tprojekte = { "projekte" };
		deleteFromTables(tables);
		deleteFromTables(tatag);
		deleteFromTables(tmitarbeiter);
		deleteFromTables(tbasis);
		deleteFromTables(tprojekte);
		Basis basis = new Basis();
		basis.setBasis_bez("test");
		basisDao.save(basis);
		Mitarbeiter mitarbeiter = new Mitarbeiter();
		mitarbeiter.setMit_name("Mustermann");
		mitarbeiter.setMit_vorname("Max");
		mitarbeiter.setMit_login("mm");
		mitarbeiter.setMit_pw("123");
		mitarbeiter.setMit_isadmin(true);
		mitarbeiterDao.save(mitarbeiter, basis.getId());
		Arbeitstag arbeitstag = new Arbeitstag();
		Calendar calendar = new GregorianCalendar();
		calendar.set(Calendar.DATE, 9);
		calendar.set(Calendar.MONTH, 9);
		calendar.set(Calendar.YEAR, 2012);
		calendar.set(Calendar.HOUR_OF_DAY, 9);
		calendar.set(Calendar.MINUTE, 0);
		calendar.set(Calendar.SECOND, 0);
		calendar.set(Calendar.MILLISECOND, 0);
		arbeitstag.setAtag_von(calendar.getTime());
		calendar.add(Calendar.HOUR_OF_DAY, 9);
		arbeitstag.setAtag_bis(calendar.getTime());
		arbeitstag.setAtag_datum(calendar.getTime());
		arbeitstag.setMitarbeiter(mitarbeiter);
		arbeitstagDao.save(arbeitstag);
		calendar.set(Calendar.HOUR_OF_DAY, 12);
		Projekte prj = new Projekte();
		prj.setProjekt_aktiv(true);
		prj.setProjekt_bez("Unittest");
		projekteDao.save(prj);
		Projektzeit prjzeit = new Projektzeit();
		prjzeit.setProjekt_von(calendar.getTime());
		calendar.add(Calendar.HOUR_OF_DAY, 1);
		if (!flag)
			prjzeit.setProjekt_bis(calendar.getTime());
		prjzeit.setArbeitstag(arbeitstag);
		prjzeit.setProjekte(prj);
		projektzeitDao.save(prjzeit);
		return prjzeit;		
	}

	@SuppressWarnings("deprecation")
	private void destroyZeit() {
		String[] tables = { "projektzeit" };
		String[] tatag = { "arbeitstag" };
		String[] tmitarbeiter = { "mitarbeiter" };
		String[] tbasis = { "basis" };
		String[] tprojekte = { "projekte" };
		deleteFromTables(tables);
		deleteFromTables(tatag);
		deleteFromTables(tmitarbeiter);
		deleteFromTables(tbasis);
		deleteFromTables(tprojekte);
	}
}
